home *** CD-ROM | disk | FTP | other *** search
/ Collection of Tools & Utilities / Collection of Tools and Utilities.iso / archiver / pdtar.zip / NDIR.C < prev    next >
C/C++ Source or Header  |  1987-08-26  |  3KB  |  181 lines
  1. /*
  2.  * MS-DOS routines to read directories in the BSD format
  3.  * By J. Eric Roskos    1/87  Public domain.
  4.  *
  5.  * Revision History:
  6.  *    3/08/87        Eliminate use of static strings; malloc search string,
  7.  *                store whichdir in DIR structure.  Make dosdir structure
  8.  *                a substructure of DIR.
  9.  */
  10.  
  11. #include "ndir.h"
  12. #include <dos.h>
  13. #include <stdio.h>
  14.  
  15. extern int      _doserrno;
  16.  
  17. char           *dos_errlst[] =
  18. {
  19.     "",
  20.     "Invalid function number",
  21.     "File not found",
  22.     "Path not found",
  23.     "Too many open files",
  24.     "Access denied",
  25.     "Invalid handle",
  26.     "Memory control blocks destroyed",
  27.     "Insufficient memory",
  28.     "Invalid memory block address",
  29.     "Invalid environment",
  30.     "Invalid format",
  31.     "Invalid access code",
  32.     "Invalid data",
  33.     "?",
  34.     "Invalid drive",
  35.     "Can't remove current directory",
  36.     "Not same device",
  37.     "No more files",
  38.     ""
  39. };
  40.  
  41. /*
  42.  * open a directory.
  43.  */
  44. DIR            *
  45. opendir(name)
  46. char           *name;
  47. {
  48.     register DIR   *dirp;
  49.     char           *malloc();
  50.  
  51.     dirp = (DIR *) malloc(sizeof(DIR));
  52.     if (dirp == NULL)
  53.     {
  54.         fprintf(stderr, "opendir: malloc failed\n");
  55.         return (NULL);
  56.     }
  57.     strncpy(dirp->whichdir, name, sizeof(dirp->whichdir));
  58.     dirp->dosdir.doshdr[0] = 'E';
  59.     dirp->dosdir.doshdr[1] = 'R';
  60.     return dirp;
  61. }
  62.  
  63.  
  64.  
  65. /*
  66.  * close a directory.
  67.  */
  68. void
  69. closedir(dirp)
  70. register DIR   *dirp;
  71. {
  72.     free(dirp);
  73. }
  74.  
  75.  
  76.  
  77. /*
  78.  * read a DOS directory entry and present it as a new Unix one
  79.  */
  80.  
  81. /*
  82.  * get next entry in a directory.
  83.  */
  84. struct direct  *
  85. readdir(dirp)
  86. register DIR   *dirp;
  87. {
  88.     static struct direct dir;
  89.     union REGS      inregs, outregs;
  90.     char           *search;
  91.     char           *malloc();
  92.  
  93.     inregs.h.ah = 0x1a;            /* set DTA */
  94.     inregs.x.dx = (int) &dirp->dosdir;
  95.     intdos(&inregs, &outregs);
  96.     if (outregs.x.cflag)
  97.     {
  98.         fprintf(stderr, "readdir: set DTA: dos error %d\n", _doserrno);
  99.         return (NULL);
  100.     }
  101.  
  102.     if (dirp->dosdir.doshdr[0] == 'E' && dirp->dosdir.doshdr[1] == 'R')
  103.         inregs.h.ah = 0x4e;        /* find first */
  104.     else
  105.         inregs.h.ah = 0x4f;        /* find next  */
  106.     search = malloc(strlen(dirp->whichdir) + 8);
  107.     if (!search)
  108.     {
  109.         fprintf(stderr, "readdir: malloc failed\n");
  110.         perror("malloc");
  111.         return (NULL);
  112.     }
  113.     sprintf(search, "%s/*.*", dirp->whichdir);
  114.     inregs.x.dx = (int) search;    /* match any name */
  115.     inregs.x.cx = 0x16;            /* match any type */
  116.     intdos(&inregs, &outregs);
  117.     free(search);
  118.     dir.d_ino = 1;
  119.     if (outregs.x.cflag)
  120.     {
  121.         if (_doserrno != 18)    /* 18 = "end of dir" */
  122.             fprintf(stderr, "readdir: '%s': %s\n", dirp->whichdir,
  123.                 dos_errlst[_doserrno]);
  124.         dir.d_ino = 0;
  125.         dir.d_name[0] = '\0';
  126.         return (NULL);
  127.     }
  128.     else
  129.         strncpy(dir.d_name, strlwr(dirp->dosdir.name), 13);
  130.     dir.d_namlen = strlen(dir.d_name);
  131.     return (&dir);
  132. }
  133.  
  134. /*
  135.  * Go back to the beginning of a directory.
  136.  */
  137.  
  138. rewinddir(dirp)
  139. DIR            *dirp;
  140. {
  141.     dirp->dosdir.doshdr[0] = 'E';
  142.     dirp->dosdir.doshdr[1] = 'R';
  143. }
  144.  
  145.  
  146. #ifdef TEST
  147.  
  148. main(argc, argv)
  149. int             argc;
  150. char          **argv;
  151. {
  152.     DIR            *dirp;
  153.     struct direct  *dir;
  154.     int             rew = 0;
  155.  
  156.     if (argc > 1)
  157.         dirp = opendir(argv[1]);
  158.     else
  159.         dirp = opendir(".");
  160.  
  161. again:
  162.  
  163.     do
  164.     {
  165.         dir = readdir(dirp);
  166.         printf("%s\n", dir->d_name);
  167.     } while (dir->d_ino);
  168.  
  169.     if (!rew++)                    /* test rewinddir once */
  170.     {
  171.         printf("**** Rewinding ****\n");
  172.         rewinddir(dirp);
  173.         goto again;
  174.     }
  175.  
  176.     closedir(dirp);
  177.     exit(0);
  178. }
  179.  
  180. #endif
  181.